Arm backend: Add initial module tests for Stable Diffusion 3.5 Medium #12242
Conversation
- Add module tests for two text encoders: CLIP and T5 - Add module tests for VAE autoencoder - Add module tests for SD3Transformer2DModel - Add flag to exmaples/arm/setup.sh for installing Stable Diffusion dependencies - Handle int64 inputs to aten.slice_copy.Tensor using pass InsertCastForOpsWithInt64InputPass Change-Id: I4389e87749cfb4e40f837cc6bfa8a59a73fb3a3a Signed-off-by: Yufeng Shi <yufeng.shi@arm.com>
YufengShi-dudu
added
partner: arm
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12242
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (3 Unrelated Failures)As of commit 53447d4 with merge base d952326 ( BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
facebook-github-bot
added
the
CLA Signed
|
Hi @digantdesai, We added some test configs to stable_diffusion_module_test_configs.py. They're mostly copy-pasted from the diffusers library to initialize different modules for testing. We've included license notice and copyright attribution, and also links to the original source files for reference, Please let us know if you have any concerns. |
Thx for highlighting this @YufengShi-dudu. With the proper attributions in place and a clear separation (separate file), it should be fine. But pls @digantdesai or @mergennachin - pls confirm. |
Change-Id: I4389e87749cfb4e40f837cc6bfa8a59a73fb3a3a
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218